package com.freedy.dataStructure.sparseArr;

import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.io.FileOutputStream;
import java.io.IOException;

/**
 * @author Freedy
 * @date 2021/3/14 11:05
 */
public class SparseArr {
    public static void main(String[] args) throws IOException {
        int[][] origin=new int[11][11];
        origin[1][4]=1;
        origin[3][3]=2;
        System.out.println("===========原始数据===========");
        printArr(origin);

        int[][] sparseArr = normalArrToSparseArr(origin);
        System.out.println("===========稀疏数组===========");
        printArr(sparseArr);

        int[][] normalArr = sparseArrToNormalArr(sparseArr);
        System.out.println("===========转化回来的普通数组===========");
        printArr(normalArr);
    }

    /**
     * 将稀疏数组转化为普通数组
     */
    public static int[][] sparseArrToNormalArr(int[][] sparseArr){
        int[][] arr=new int[sparseArr[0][0]][sparseArr[0][1]];
        for (int i = 1; i < sparseArr[0][2]+1; i++) {
            arr[sparseArr[i][0]][sparseArr[i][1]]=sparseArr[i][2];
        }
        return arr;
    }


    /**
     * 将普通数组转稀疏数组
     */
    public static int[][] normalArrToSparseArr(int[][] origin){
        //统计非零数据的个数，方便之后创建稀疏数组
        int sum=0;
        //先遍历二维数组得到非0的数据
        for (int i = 0; i < origin.length; i++) {
            for (int j = 0; j < origin[0].length; j++) {
                if (origin[i][j]!=0){
                    sum++;
                }
            }
        }
        int[][] sparseArr=new int[sum+1][3];
        //对稀疏数组赋初始值
        sparseArr[0][0]=origin.length;
        sparseArr[0][1]=origin[0].length;
        sparseArr[0][2]=sum;
        //对稀疏数组赋数据值
        int row=1;
        for (int i = 0; i < origin.length; i++) {
            for (int j = 0; j < origin[0].length; j++) {
                if (origin[i][j]!=0){
                    sparseArr[row][0]=i;
                    sparseArr[row][1]=j;
                    sparseArr[row][2]=origin[i][j];
                    row++;
                }
            }
        }
       return sparseArr;
    }


    /**
     * 打印二维数组
     * @param a
     */
    public static void printArr(int[][] a){
        for (int[] ints : a) {
            for (int anInt : ints) {
                System.out.printf("%d\t",anInt);
            }
            System.out.println();
        }
    }
}
